NumPy Polyfit function in Python